package com.chinanet.service.faultanticipate.swapservice;

import org.apache.commons.codec.binary.Base64;

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;


/**
 * Created by moon on 16/10/14.
 */
public class swapservice {
    public static final String Key = "Esw349SIZ74js05N";

    public static void main(String args[]){
        String str = getToken();
//        for(int i=0;i<100;i++){
            System.out.println(getToken());
//        }
//        if(comparToken(str)){
//            System.out.println("yes");
//        }else {
//            System.out.println("no");
//        }
    }

    public static String getToken(){
        String begin = ""+System.currentTimeMillis();
        // 加密
        try {
            String enString = Encrypt(begin, Key);
            return enString;
        }catch (Exception e){
            e.printStackTrace();
        }
        // 解密
        return null;
    }

    public static String comparToken(String token) {
        try {
            String DeString = Decrypt(token, Key);
            String tp[] = DeString.split("\\|");
            if(tp.length==2){
                long l = Long.parseLong(tp[1]);
                if(l-System.currentTimeMillis()<16000){
                    return tp[0];
                }else {
                    return null;
                }
            }else {
                return null;
            }
        }catch (Exception e){
            e.printStackTrace();
            return null;
        }
    }

    // 加密
    public static String Encrypt(String sSrc, String sKey) throws Exception {
        if (sKey == null) {
            System.out.print("Key为空null");
            return null;
        }
        // 判断Key是否为16位
        if (sKey.length() != 16) {
            System.out.print("Key长度不是16位");
            return null;
        }
        byte[] raw = sKey.getBytes("utf-8");
        SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
        Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");//"算法/模式/补码方式"
        cipher.init(Cipher.ENCRYPT_MODE, skeySpec);
        byte[] encrypted = cipher.doFinal(sSrc.getBytes("utf-8"));

        return new Base64().encodeToString(encrypted);//此处使用BASE64做转码功能，同时能起到2次加密的作用。
    }

    // 解密
    public static String Decrypt(String sSrc, String sKey) throws Exception {
        try {
            // 判断Key是否正确
            if (sKey == null) {
                return null;
            }
            // 判断Key是否为16位
            if (sKey.length() != 16) {
                return null;
            }
            byte[] raw = sKey.getBytes("utf-8");
            SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
            Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
            cipher.init(Cipher.DECRYPT_MODE, skeySpec);
            byte[] encrypted1 = new Base64().decode(sSrc);//先用base64解密
            try {
                byte[] original = cipher.doFinal(encrypted1);
                String originalString = new String(original,"utf-8");
                return originalString;
            } catch (Exception e) {
                System.out.println(e.toString());
                return null;
            }
        } catch (Exception ex) {
            System.out.println(ex.toString());
            return null;
        }
    }


}
